package week_05;

import java.util.*;

/**
 * 56. 合并区间
 * 以数组 intervals 表示若干个区间的集合，其中单个区间为 intervals[i] = [starti, endi] 。
 * 请你合并所有重叠的区间，并返回一个不重叠的区间数组，该数组需恰好覆盖输入中的所有区间。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：intervals = [[1,3],[2,6],[8,10],[15,18]]
 * 输出：[[1,6],[8,10],[15,18]]
 * 解释：区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6].
 * 示例 2：
 * <p>
 * 输入：intervals = [[1,4],[4,5]]
 * 输出：[[1,5]]
 * 解释：区间 [1,4] 和 [4,5] 可被视为重叠区间。
 */
public class MergeIntervals_1_56 {

    public int[][] merge(int[][] intervals) {
        // 先求sort
        List<Item> list = new ArrayList<>();
        for (int[] arr : intervals) {
            list.add(new Item(arr[0], arr[1]));
        }
        Collections.sort(list, new Comparator<Item>() {
            @Override
            public int compare(Item o1, Item o2) {
                if (o1.k1 != o2.k1) {
                    return o1.k1 - o2.k1;
                } else {
                    return o1.k2 - o2.k2;
                }
            }
        });

        List<Item> ans = new ArrayList<>();
        int left = list.get(0).k1;
        int far = list.get(0).k2;
        for (Item item : list) {
            int start = item.k1;
            int end = item.k2;
            if (start <= far) {
                far = Math.max(far, end);
            } else {
                if (far > -1) ans.add(new Item(left, far));
                left = start;
                far = end;
            }
        }
        if (far > -1) ans.add(new Item(left, far));
        return ans.stream().map(u -> new int[]{u.k1, u.k2}).toArray(int[][]::new);
    }

    public int[][] merge2(int[][] intervals) {
        // 先求sort
        List<Item> list = new ArrayList<>();
        for (int[] arr : intervals) {
            list.add(new Item(arr[0], arr[1]));
        }
        Collections.sort(list, new Comparator<Item>() {
            @Override
            public int compare(Item o1, Item o2) {
                if (o1.k1 != o2.k1) {
                    return o1.k1 - o2.k1;
                } else {
                    return o1.k2 - o2.k2;
                }
            }
        });

        int[][] ans = new int[intervals.length][2];
        int start = list.get(0).k1;
        int end = list.get(0).k2;
        int index = 0;
        for (Item item : list) {
            if (end < item.k1) {
                // 新的一段
                ans[index] = new int[]{start, end};
                index++;
                start = item.k1;
                end = item.k2;
            } else {
                if (end < item.k2) end = item.k2;
            }
        }
        ans[index] = new int[]{start, end};

        return Arrays.copyOf(ans, index + 1);
    }

    public class Item {
        int k1;
        int k2;

        public Item(int k1, int k2) {
            this.k1 = k1;
            this.k2 = k2;
        }
    }

    public static void main(String[] args) {
        MergeIntervals_1_56 test = new MergeIntervals_1_56();
        int[][] intervals = new int[][]{{1, 3}, {2, 6}, {8, 10}, {15, 18}};
        test.merge(intervals);
    }
}
